package com.qf.electronic.handler;

import com.qf.electronic.common.R;
import com.qf.electronic.common.ResponseStatus;
import com.qf.electronic.util.ResponseUtil;
import org.springframework.security.authentication.DisabledException;
import org.springframework.security.core.AuthenticationException;
import org.springframework.security.core.userdetails.UsernameNotFoundException;
import org.springframework.security.web.authentication.AuthenticationFailureHandler;
import org.springframework.stereotype.Component;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;

@Component
public class LoginFailureHandler implements AuthenticationFailureHandler {


    @Override
    public void onAuthenticationFailure(HttpServletRequest request,
                                        HttpServletResponse response,
                                        AuthenticationException exception)
            throws IOException, ServletException {
        if(exception instanceof UsernameNotFoundException){
            ResponseUtil.writeResponse(response, R.error(ResponseStatus.NO_USER));
        } else if(exception instanceof DisabledException){
            ResponseUtil.writeResponse(response, R.error(ResponseStatus.USER_DISABLED));
        } else {
            ResponseUtil.writeResponse(response, R.error(ResponseStatus.BAD_CREDENTIALS));
        }
    }
}
